# 模型的单个测试

img = learn.data.train_ds[0][0]
learn.predict(img)
# 查看每个类的概率
np.exp(out[2].numpy())